Pytorch 1.8.1
Python 3.8.8
Torchvision 0.9.1
numpy

